How can I fix the TypeError: 'str' object cannot be interpreted as an integer in my Python code